#!/usr/bin/env bash
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
export PYTHONPATH=${DIR}/../../../:${PYTHONPATH}
export PYTHONPATH=${DIR}/../../../slim:${PYTHONPATH}
export PYTHONPATH=${DIR}/../../:${PYTHONPATH}

pretrained=${DIR}/../../data/ssd_mobilenet_v1_coco_11_06_2017
config=${DIR}/../../samples/configs/ssd_mobilenet_beer.config
output=${pretrained}/$1

DATA=/home/admins/data/beer_data/pre

python3 ${DIR}/../../export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path ${config} \
--trained_checkpoint_prefix ${pretrained}/model.ckpt- \
--output_directory ${pretrained}/
python3 prediction.py